(Archive)ComfyUI WD 1.4 Tagger

モデルのダウンロード

性能は多分どれもそんなに変わらない

https://huggingface.co/SmilingWolf/wd-convnext-tagger-v3/tree/mainwd-convnext-tagger-v3

https://huggingface.co/SmilingWolf/wd-swinv2-tagger-v3/tree/mainwd-swinv2-tagger-v3

https://huggingface.co/SmilingWolf/wd-vit-tagger-v3/tree/mainwd-vit-tagger-v3

model.onnxとselected_tags.csvをダウンロードしてpath\to\ComfyUI\custom_nodes\ComfyUI-WD14-Tagger\modelsに置く

それぞれのファイル名を変えておく

e.g. wd-convnext-tagger-v3.onnx / wd-convnext-tagger-v3.csv

wd14tagger.pyを書き換える

$ path\to\ComfyUI\custom_nodes\ComfyUI-WD14-Tagger\wd14tagger.py

32行目

code:diff

- all_models = ("wd-v1-4-moat-tagger-v2", "wd-v1-4-convnext-tagger-v2", "wd-v1-4-convnext-tagger","wd-v1-4-convnextv2-tagger-v2", "wd-v1-4-vit-tagger-v2")

+ all_models = ("wd-v1-4-moat-tagger-v2", "wd-v1-4-convnext-tagger-v2", "wd-v1-4-convnext-tagger", "wd-v1-4-convnextv2-tagger-v2", "wd-v1-4-vit-tagger-v2","wd-vit-tagger-v3","wd-swinv2-tagger-v3","wd-convnext-tagger-v3")

#Archive_page